# encoding: utf-8

import torch
import torch.nn as nn
from torch.utils import data
from PIL import Image
import os
from glob import glob





# batch_size = 1
# datasets = MyDataSet("./data/train")
# dataloader = data.DataLoader(datasets, batch_size=batch_size)
#
# for i, data in enumerate(dataloader):
#     print(i, data)
#
#     break

